package com.javaee.wordtree.controller;

import com.javaee.wordtree.common.Result;
import com.javaee.wordtree.entity.Chat;
import com.javaee.wordtree.service.ChatService;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.List;
import java.util.Map;

/**
 * 聊天控制器
 */
@RestController
@RequestMapping("/chats")
public class ChatController {
    
    @Resource
    private ChatService chatService;
    
    /**
     * 获取用户的聊天列表
     * @return 聊天列表
     */
    @GetMapping
    public Result<?> getUserChats(HttpServletRequest request) {
        // 在实际应用中，用户ID应该从JWT或session中获取
        String userId = getCurrentUserId(request);
        List<Chat> chats = chatService.getUserChats(userId);
        return Result.success(chats);
    }
    
    /**
     * 创建新的聊天会话
     * @param params 请求参数
     * @return 创建的聊天会话
     */
    @PostMapping
    public Result<?> createChat(@RequestBody Map<String, String> params, HttpServletRequest request) {
        String userId = getCurrentUserId(request);
        String title = params.getOrDefault("title", "新的会话");
        Chat chat = chatService.createChat(userId, title);
        return Result.success(chat);
    }
    
    /**
     * 更新聊天标题
     * @param chatId 聊天ID
     * @param params 请求参数
     * @return 操作结果
     */
    @PutMapping("/{chatId}")
    public Result<?> updateChat(@PathVariable Integer chatId, 
                            @RequestBody Map<String, String> params, 
                            HttpServletRequest request) {
        String userId = getCurrentUserId(request);
        
        // 检查聊天是否属于当前用户
        if (!chatService.isChatBelongToUser(chatId, userId)) {
            return Result.error("-1", "没有权限操作此聊天记录");
        }
        
        String title = params.get("title");
        if (title == null || title.isEmpty()) {
            return Result.error("-1", "标题不能为空");
        }
        
        boolean success = chatService.updateChatTitle(chatId, title);
        return success ? Result.success() : Result.error("-1", "更新失败");
    }
    
    // /**
    //  * 标记聊天为已读
    //  * @param chatId 聊天ID
    //  * @return 操作结果
    //  */
    // @PutMapping("/{chatId}/read")
    // public Result<?> markAsRead(@PathVariable Integer chatId, HttpServletRequest request) {
    //     String userId = getCurrentUserId(request);
        
    //     // 检查聊天是否属于当前用户
    //     if (!chatService.isChatBelongToUser(chatId, userId)) {
    //         return Result.error("-1", "没有权限操作此聊天记录");
    //     }
        
    //     // 标记为已读的逻辑，在当前示例中不需要特殊处理
    //     return Result.success();
    // }
    
    /**
     * 删除聊天记录
     * @param chatId 聊天ID
     * @return 操作结果
     */
    @DeleteMapping("/{chatId}")
    public Result<?> deleteChat(@PathVariable Integer chatId, HttpServletRequest request) {
        String userId = getCurrentUserId(request);
        
        // 检查聊天是否属于当前用户
        if (!chatService.isChatBelongToUser(chatId, userId)) {
            return Result.error("-1", "没有权限操作此聊天记录");
        }
        
        boolean success = chatService.deleteChat(chatId);
        return success ? Result.success() : Result.error("-1", "删除失败");
    }
    
    /**
     * 获取当前用户ID（实际应用中应该从JWT或session中获取）
     */
    private String getCurrentUserId(HttpServletRequest request) {
        // 这里简单模拟从请求头中获取用户ID
        String userId = request.getHeader("X-User-ID");
        // 临时使用固定用户ID进行开发测试
        return userId != null ? userId : "123456";
    }
}
